package com.dubbo.mysql;


import java.sql.Connection;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;
import javax.sql.DataSource;
import org.apache.commons.dbcp2.BasicDataSource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
 *       dbcp 实用类，提供了dbcp连接，不允许继承； * 
 *       该类需要有个地方来初始化 DS ，通过调用initDS 方法来完成， 可以在通过调用带参数的构造函数完成调用，
 *       可以在其它类中调用，也可以在本类中加一个static{}来完成；
 */
public final class DbcpConPool {

	/** 数据源，static */
	private static DataSource DS;
	private final static Logger log = LoggerFactory.getLogger(Distrmysql.class);
	/** 从数据源获得一个连接 */
	public Connection getConn() {
		Connection con = null;
		if (DS != null) {
			try {
				con = DS.getConnection();
			} catch (Exception e) {
				e.printStackTrace(System.err);
			}

			try {
				if(con != null)
				con.setAutoCommit(false);
			} catch (SQLException e) {
				log.error("get{}", e.getMessage());
			}
			return con;
		}
		return con;
	}

	/** 默认的构造函数 */
	public DbcpConPool() {
	}

	/** 构造函数，初始化了 DS ，指定 数据库 */
	public DbcpConPool(String connectURI) {
		initDS(connectURI);
	}

	/** 构造函数，初始化了 DS ，指定 所有参数 */
	public DbcpConPool(String connectURI, String username, String pswd,
			String driverClass) {
		initDS(connectURI, username, pswd, driverClass);
	}

	/**
	 * 创建数据源，除了数据库外，都使用硬编码默认参数；
	 * 
	 * @param connectURI
	 *            数据库
	 *
	 * uqury
	 */
	public static void initDS(String connectURI) {
		initDS(connectURI, "uqury", "1qaz@WSX", "com.mysql.jdbc.Driver");
	}

	/**
	 * 指定所有参数连接数据源
	 * 
	 * @param connectURI
	 *            数据库
	 * @param username
	 *            用户名
	 * @param pswd
	 *            密码
	 * @param driverClass
	 *            数据库连接驱动名
	 * initialSize
	 *            初始连接池连接个数
	 * maxtotal
	 *            最大活动连接数
	 * maxIdle
	 *            最大连接数
	 * maxWaitMillis
	 *            获得连接的最大等待毫秒数
	 * minIdle
	 *            最小连接数
	 * @return
	 */
	public static void initDS(String connectURI, String username, String pswd,
			String driverClass) {
		BasicDataSource ds = new BasicDataSource();
		ds.setTestOnBorrow(false);
		ds.setTestOnReturn(false);
		ds.setDriverClassName(driverClass);
		ds.setUsername(username);
		ds.setPassword(pswd);
		ds.setUrl(connectURI);
		ds.setInitialSize(10); // 初始的连接数；
		ds.setMaxTotal(80);
		ds.setMaxIdle(20);
		ds.setMaxWaitMillis(500);
		ds.setMinIdle(1);
		DS = ds;
	}

	/** 获得数据源连接状态 */
	public static Map<String, Integer> getDataSourceStats() throws SQLException {
		BasicDataSource bds = (BasicDataSource) DS;
		Map<String, Integer> map = new HashMap<String, Integer>(2);
		map.put("active_number", bds.getNumActive());
		map.put("idle_number", bds.getNumIdle());
		return map;
	}

	/** 关闭数据源 */
	protected static void shutdownDataSource() throws SQLException {
		BasicDataSource bds = (BasicDataSource) DS;
		bds.close();
	}
}